package com.hmdp.service.impl;

import cn.hutool.Hutool;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.lang.UUID;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hmdp.dto.LoginFormDTO;
import com.hmdp.dto.Result;
import com.hmdp.dto.UserDTO;
import com.hmdp.entity.User;
import com.hmdp.mapper.UserMapper;
import com.hmdp.service.IUserService;
import com.hmdp.utils.RedisConstants;
import com.hmdp.utils.RegexUtils;
import com.hmdp.utils.SystemConstants;
import com.hmdp.utils.UserHolder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.data.redis.connection.BitFieldSubCommands;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import javax.servlet.http.HttpSession;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import static com.hmdp.utils.RedisConstants.*;

/**
 * <p>
 * 服务实现类
 * </p>
 *
 * @author 虎哥
 * @since 2021-12-22
 */
@Service
@Slf4j
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements IUserService {

    @Resource
    private RedisTemplate<String, Object> redisTemplate;
    @Resource(name = "stringRedisTemplate")
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 手机号校验，生成验证码并保存到session，(发送验证码)
     * @param phone
     * @param session
     * @return
     */
    @Override
    public Result sendCode(String phone, HttpSession session) {
        // 0.校验手机号
        if (RegexUtils.isPhoneInvalid(phone)) {
            // 2.如果不符合，返回错误信息
            return Result.fail("手机号格式错误！");
        }

        // 1. 判断是否在一级限制条件内
        Boolean oneLevelLimit = stringRedisTemplate.opsForSet().isMember(ONE_LEVERLIMIT_KEY + phone, "1");//Check if set at key contains value.
        if (oneLevelLimit != null && oneLevelLimit) {
            // 在一级限制条件内，不能发送验证码
            return Result.fail("您需要等5分钟后再请求");
        }

// 3. 检查过去1分钟内发送验证码的次数
        long oneMinuteAgo = System.currentTimeMillis() - 60 * 1000;
        //Count number of elements within sorted set with scores between min and max
        //计数：score介于一分钟前时间和当前时间之间的元素数量
        long count_oneminute = stringRedisTemplate.opsForZSet().count(SENDCODE_SENDTIME_KEY + phone, oneMinuteAgo, System.currentTimeMillis());
        if (count_oneminute >= 1) {
            // 过去1分钟内已经发送了1次，不能再发送验证码
            return Result.fail("距离上次发送时间不足1分钟，请1分钟后重试");
        }

        // 4. 检查发送验证码的次数
        long fiveMinutesAgo = System.currentTimeMillis() - 5 * 60 * 1000;
        //计数：score介于五分钟前时间和当前时间之间的元素数量
        long count_fiveminute = stringRedisTemplate.opsForZSet().count(SENDCODE_SENDTIME_KEY + phone, fiveMinutesAgo, System.currentTimeMillis());
        if (count_fiveminute == 5) {
            // 过去5分钟内已经发送了5次，进入一级限制
            stringRedisTemplate.opsForSet().add(ONE_LEVERLIMIT_KEY + phone, "1");
            stringRedisTemplate.expire(ONE_LEVERLIMIT_KEY + phone, 5, TimeUnit.MINUTES);
            return Result.fail("5分钟内已经发送了5次，接下来如需再发送请等待5分钟后重试");
        }

        //生成验证码
        String code = RandomUtil.randomNumbers(6);

        //将生成的验证码保持到redis
        stringRedisTemplate.opsForValue().set(LOGIN_CODE_KEY+phone,code,LOGIN_CODE_TTL, TimeUnit.MINUTES);

        log.info("发送登录验证码：{}", code);

        // 更新发送时间和次数 key:phone value:curTime score:curTime
        stringRedisTemplate.opsForZSet().add(SENDCODE_SENDTIME_KEY + phone, System.currentTimeMillis() + "", System.currentTimeMillis());

        return Result.ok();
    }

    /**
     * 登录/快速注册
     * @param loginForm 登录参数，包含手机号、验证码；或者手机号、密码
     * @param session
     * @return
     */
    @Override
    public Result login(LoginFormDTO loginForm, HttpSession session) {
        //1. 校验手机号
        String phone = loginForm.getPhone();
        if (RegexUtils.isPhoneInvalid(phone)) {
            return Result.fail("手机号格式错误！");
        }
        log.info("================手机号格式正确...============");
        //TODO 压测调试时，不进行验证码校验
//        //2. 校验验证码
//        String code = loginForm.getCode();
////        String sessionCode = (String) session.getAttribute("code");//session中保存的验证码
//        //redis中保存的验证码
//        String redisCode = (String) redisTemplate.opsForValue().get(RedisConstants.LOGIN_CODE_KEY + phone);
//        if (StrUtil.isEmpty(redisCode) || !redisCode.equals(code)) {
//            return Result.fail("验证码错误！");
//        }

        //3. 根据手机号查询用户信息 select * from tb_user where phone = "xxx"
        User dbUser = query().eq("phone", phone).one();//数据库查询

        //4. 用户信息不存在，是注册操作，向数据库表tb_user插入用户数据
        if (dbUser == null) {
            dbUser = createUserWithPhone(phone);
        }
//        //5. 将用户数据保存到session 没有必要存user所有属性：信息泄露/增大存储压力 解决：使用UserDTO代替User
//        UserDTO userDTO = new UserDTO();
//        BeanUtils.copyProperties(dbUser, userDTO);
//        session.setAttribute("user", userDTO);

        //5. 将用户数据保存到redis
        UserDTO userDTO = new UserDTO();
        BeanUtils.copyProperties(dbUser, userDTO);
        //5.1 随机生成token作为令牌(redis key)
        String token = UUID.randomUUID().toString(true);
        //5.2 User信息保存使用Hash结构 userDTO->MAP->存储
        /*
        用户信息有效期的含义：应保证每次发起请求，都设置一次30min的有效期，超过30min
        无请求时，用户信息过期，而不是仅仅在登陆时设置30min的有效期。
        只要客户端发起请求，就不断更新用户信息有效期，保证登录状态->在拦截器中做
         */
        Map<String, Object> userMap = BeanUtil.beanToMap(userDTO);
        redisTemplate.opsForHash().putAll(RedisConstants.LOGIN_USER_KEY + token, userMap);
        //TODO 永不过期，方便调试
        //redisTemplate.expire(RedisConstants.LOGIN_USER_KEY + token, RedisConstants.LOGIN_USER_TTL, TimeUnit.MINUTES);//有效期30min

        //5.3 将token返回给前端，以便后续发起请求时携带
        return Result.ok(token);
    }

    /**
     * 用户签到
     * @return
     */
    @Override
    public Result sign() {
        //获取当前用户
        Long userId = UserHolder.getUser().getId();
        //获取日期：年月日
        String dateStr = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
        String[] dateSubStrs = dateStr.split("-");
        String year = dateSubStrs[0];
        String month = dateSubStrs[1];
        String day = dateSubStrs[2];
        //拼接key：用户+年月
        String redisKey = RedisConstants.USER_SIGN_KEY + userId + ":" + year + ":" + month;
        //获取今天是本月的第几天(从0开始)
        int dayOfMonth = day.charAt(0) == '0'? Integer.parseInt(day.substring(1)) : Integer.parseInt(day);
        dayOfMonth -= 1;
        //写入redis bitMap
        stringRedisTemplate.opsForValue().setBit(redisKey, dayOfMonth, true);
        return Result.ok();
    }

    /**
     * 获取当前用户本月截至今天的连续签到天数
     * @return
     */
    @Override
    public Result signCount() {
        //获取当前用户
        Long userId = UserHolder.getUser().getId();
        //获取日期：年月
        String dateStr = LocalDateTime.now().format(DateTimeFormatter.ofPattern(":yyyy:MM"));
        //拼接key：用户+年月
        String redisKey = RedisConstants.USER_SIGN_KEY + userId + dateStr;
        //获取今天是本月的第几天(从0开始)
        int dayOfMonth = LocalDateTime.now().getDayOfMonth();
        //本月截至今天的所有签到记录，返回十进制数字 BITFIELD key GET u18 0 (u-返回无符号数字，18-查18个位，0从第0位开始)
        List<Long> signRecords = stringRedisTemplate.opsForValue().bitField(redisKey,
                BitFieldSubCommands.create()
                        .get(BitFieldSubCommands.BitFieldType.unsigned(dayOfMonth)).valueAt(0));
        if (signRecords == null || signRecords.isEmpty()) {
            return Result.ok(0);
        }
        Long signRecord = signRecords.get(0);
        if (signRecord == null || signRecord == 0) {
            return Result.ok(0);
        }
        //循环遍历，让这个数字与1做与运算，右移，重复操作，可不断获取数字的最后一位
        int signCounter = 0;
        for (int bitCnt = 0; bitCnt < dayOfMonth; bitCnt++) {
            long lastBit = signRecord & 1;
            if (lastBit == 0) {
                //未签到，结束统计
                break;
            }
            //已签到，签到数加1
            signCounter++;
            //右移，统计下一位
            signRecord >>= 1;
        }
        return Result.ok(signCounter);
    }

    /**
     * 创建电话为phone的User对象，并将一条记录插入tb_user表，返回该User对象
     * @param phone
     * @return
     */
    private User createUserWithPhone(String phone) {
        User user = new User();
        user.setPhone(phone);
        user.setNickName(SystemConstants.USER_NICK_NAME_PREFIX + RandomUtil.randomString(10));//设置初始昵称：随机字符串
        save(user);//数据库插入
        return user;
    }
}
